package com.xzj.admin.analysis.base.util;

import com.alibaba.fastjson2.JSONArray;
import com.alibaba.fastjson2.JSONObject;
import org.dom4j.*;

import java.util.List;

/**
 * @program: xzj-admin
 * @ClassName: XmlToJsonUtil
 * @description: xml转Json方法
 * @author: XiaZunJun
 * @create: 2024-11-11 16:58
 */
public class XmlToJsonUtil {
    /**
     * 自定义
     */
    public static JSONObject startXMLToJSON(String xml) {
        //1、定义JSON对象保存结果
        JSONObject result = new JSONObject();
        try {
            //2、使用DocumentHelper.parseText()转换为DOM文档对象
            Document document = DocumentHelper.parseText(xml);
            //3、获取DOM文档根节点
            Element rootElement = document.getRootElement();
            //4、调用自定义的方法转换为JSON数据格式
            parseJson(rootElement, result);
        } catch (DocumentException e) {
            e.printStackTrace();
        }
        return result;
    }

    public static void parseJson(Element element, JSONObject result) {
        //得到所有节点属性和属性值的对象
        List<Attribute> attributes = element.attributes();
        for (Attribute attribute : attributes) {
            result.put(attribute.getName(), attribute.getValue());
        }
        //1、获取子节点列表
        List<Element> elements = element.elements();
        //2、循环子节点列表获取数据
        for (Element e : elements) {
            //3、有数据则获取
            if (!e.elements().isEmpty() && e.getText().isEmpty()) {
                //4、定义另一个JSON对象保存子节点JSON数据
                JSONObject cjson = new JSONObject();
                //5、此处调用自身继续方法继续循环取值，知道遍历完所有字节点数据
                parseJson(e, cjson);
                if (!cjson.isEmpty()) {
                    if (result.containsKey(e.getName())) {
                        JSONArray jsonArray = result.getJSONArray(e.getName());
                        jsonArray.add(cjson);
                        result.put(e.getName(), jsonArray);
                    } else {
                        result.put(e.getName(), cjson);
                    }
                    //6、添加到JSON对象
                }
            } else {
                if (!e.getText().isEmpty()) {
                    //6、添加到JSON对象
                    result.put(e.getName(), e.getText());
                }
            }
        }
    }
}
